package cn.wuwenyao.db.doc.generator.dto;

import cn.wuwenyao.db.doc.generator.entity.TableInfo;
import lombok.Data;
import org.springframework.util.CollectionUtils;

import java.util.Comparator;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

/**
 * 模块与关联表dto
 *
 * @author Bullion.Yan
 * @date 2022/04/09
 */
@Data
public class ModuleTableDto {
    public ModuleTableDto(String moduleName, List<TableInfo> tableInfoList) {
        this.moduleName = moduleName;
        this.tableInfoList = tableInfoList;
    }

    /**
     * 模块名称
     */
    private String moduleName;

    /**
     * 模块包含的表
     */
    List<TableInfo> tableInfoList;

    public static List<ModuleTableDto> getListByTableInfoList(List<TableInfo> tableInfoList){
        if(CollectionUtils.isEmpty(tableInfoList)){
           return null;
        }
        Map<String, List<TableInfo>> moduleTableMap = tableInfoList.stream().collect(
                Collectors.groupingBy(TableInfo -> TableInfo.getTableName().indexOf("_")==-1?"其它":TableInfo.getTableName().substring(0,TableInfo.getTableName().indexOf("_"))));

        List<ModuleTableDto> moduleTableDtoList = moduleTableMap.entrySet().stream().sorted(Comparator.comparing(e -> e.getKey()))
                .map(e -> new ModuleTableDto(e.getKey(), e.getValue())).collect(Collectors.toList());
        return moduleTableDtoList;
    }
}
